package com.changgou.file.controller;

import com.changgou.entity.Result;
import com.changgou.entity.StatusCode;
import com.changgou.file.util.FastDFSClient;
import com.changgou.file.util.FastDFSFile;
import org.apache.commons.lang.StringUtils;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

/**
 * @Author: HuangFu
 * @Date: 2020/1/16 10:46
 * @Description:
 */
@RestController
@RequestMapping("/file")
public class FileController {

    @PostMapping("/upload")
    public Result uploadFile(MultipartFile file){
        try{
            //判断文件是否存在
            if(file ==null){
                throw new RuntimeException("文件不存在");
            }
            //获取文件的完整路径
            String originalFilename = file.getOriginalFilename();
            if(StringUtils.isEmpty(originalFilename)){
                throw new RuntimeException("文件不存在");
            }
            //获取文件扩展名无需带.
            String extName = originalFilename.substring(originalFilename.lastIndexOf("." )+ 1);
            //获取文件内容 byte数组
            byte[] content = file.getBytes();
            //封装文件实体
            FastDFSFile fastDFSFile = new FastDFSFile(originalFilename,content,extName);
            String[] uploadResult = FastDFSClient.upload(fastDFSFile);
            //获取返回数据结果
            String url = FastDFSClient.getTrackerUrl() + "/" + uploadResult[0] + "/" + uploadResult[1];
            return new  Result(true, StatusCode.OK,"文件上传成功",url);
        }catch(Exception e){
            e.printStackTrace();
        }
        return new Result(false, StatusCode.ERROR,"文件上传失败");
    }
}
